FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime


COPY ./ /workspace/ 

RUN apt-get update && apt-get install -y libgl1 libglib2.0-0
RUN pip install flask opencv-python pandas ultralytics
 
# 启动入口
ENTRYPOINT ["python"]
CMD ["web_app.py"]

